package network;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.Socket;
import java.net.SocketException;

/**
 * Created with IntelliJ IDEA.
 * Description:基于UDP Socket 实现的回显服务器
 * User: 邓大帅
 * Date: 2024-01-27
 * Time: 15:23
 * k 千
 * m 百万
 * g 十亿
 */
public class UdpEchoServer {
//    需要先定义一个Socket对象.
//    通过网络通信，必须要使用Socket对象
    private DatagramSocket socket = null;
//    一个端口不一定能绑定成功，若是已经被占用，则会抛异常
    public UdpEchoServer(int port) throws SocketException {
        socket = new DatagramSocket(port);
    }
//    启动服务器
    public void start() throws IOException {
        System.out.println("服务器启动！");
        while (true) {
//            每次循环做三件事
//            1.读取请求并解析
//            这里的length指的是数据包一次读取数据的长度
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096],4096);
//            如果此时没有客户端请求，receive就会阻塞等待，直到有请求为止
            socket.receive(requestPacket);
//            这里的length指的是现在数组的长度
            String request = new String(requestPacket.getData(),0,requestPacket.getLength());
//            为了方便处理这个请求，把数据包转成
//            2.根据此请求计算响应(这步现在省略)
            String response = process(request);
//            3.把响应结果返回给客户端
            byte[] bytes = response.getBytes();
            DatagramPacket responsePacket = new DatagramPacket(bytes,bytes.length,requestPacket.getSocketAddress());
            socket.send(responsePacket);
//            日志信息
            System.out.printf("[%s:%d] req: %s res: %s\n", requestPacket.getAddress().toString(),
                    requestPacket.getPort(), request, response);
        }
    }
//这个方法是处理请求的，现在程序是一个回显，所以处理逻辑就是请求是啥，响应就是啥
    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer udpEchoServer = new UdpEchoServer(9090);
        udpEchoServer.start();
    }
}
